use openzeppelin_testing::{EventSpyExt, EventSpyQueue as EventSpy, ExpectedEvent};
use starknet::ContractAddress;

#[generate_trait]
pub impl VestingSpyHelpersImpl of VestingSpyHelpers {
    fn assert_only_event_amount_released(
        ref self: EventSpy, contract: ContractAddress, token: ContractAddress, amount: u256,
    ) {
        let expected = ExpectedEvent::new()
            .key(selector!("AmountReleased"))
            .key(token)
            .data(amount);
        self.assert_only_event(contract, expected);
    }
}
